Single line input in python